import tensorflow as tf
from tensorflow.keras import layers, optimizers


class DQN(tf.keras.Model):
    def __init__(self, n_feature,n_action):
        super(DQN, self).__init__()
        model = tf.keras.Sequential(
            [
                layers.Input([None,n_feature],name='s'),
                layers.Dense(20,activation='relu',name='l1'),
                layers.Dense(n_action,activation=None,name='l2')
            ]
        )
